from zhipuai import ZhipuAI


def ai(msg):
    client = ZhipuAI(api_key="a6df69db348269042b6fc9cb64d92ad5.JSCKkZMvUQbx616Z")
    response = client.chat.completions.create(
        model="glm-4",  # 填写需要调用的模型名称
        messages=[
            # {"role": "system",
            #  "content": "学习笔记助手，你的任务是为用户提供专业、准确、有见地的建议，和一份便于记忆的笔记大纲"},
            {"role": "user",
             "content": msg},  #userinput
        ],
        tools=[
            {
                "type": "retrieval",
                "retrieval": {
                    "knowledge_id": "1767941095351857152",
                    "prompt_template":"从文档\n\"\"\"\n{{knowledge}}\n\"\"\"\n中找问题\n\"\"\"\n{{question}}\n\"\"\"\n的答案,找到答案就使用文档中的内容回答问题并且可以做一些优化，找不到答案就用自身知识回答。不要复述问题，直接开始回答。不用主动的做自我介绍除非用户询问并且你要牢记你是linter网站的ai用于辅助他人学习 并且不用说出你的知识来源于在哪里"
                }
            }
        ],
        stream=True,
    )
    return response